package _11_整理题目._4_动态规划._打家劫舍;

import org.junit.Test;

import java.util.Arrays;

/**
 * 房屋成环 你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。
 * 这个地方所有的房屋都 围成一圈 ，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，
 * 相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警 。
 *
 * 给定一个代表每个房屋存放金额的非负整数数组，计算你 在不触动警报装置的情况下 ，
 * 今晚能够偷窃到的最高金额。
 *
 * 输入：nums = [2,3,2]
 * 输出：3
 * 解释：你不能先偷窃 1 号房屋（金额 = 2），然后偷窃 3 号房屋（金额 = 2）,
 * 因为他们是相邻的。
 *
 *
 */
public class _02_打家劫舍_2 {

    public int rob(int[] nums) {
        int n = nums.length;
        if(n == 1) return nums[0];
        if(n == 2) return Math.max(nums[0], nums[1]);
        return Math.max(
                robRange(Arrays.copyOfRange(nums,0,n-1)),
                robRange(Arrays.copyOfRange(nums,1,n)));
    }

    public int robRange(int[] nums) {
        int n = nums.length;
        if(n == 1) return nums[0];
        if(n == 2) return Math.max(nums[0], nums[1]);
        int a = nums[0];
        int b = Math.max(nums[0], nums[1]);
        int c = 0;
        for(int i=2; i<n; i++) {
            c = Math.max(b, a+nums[i]);
            a = b;
            b = c;
        }
        return c;
    }

    @Test
    public void main() {
        int[] nums = {2,7,9,3,1};
        System.out.println(rob(nums));
    }
}
